package com.tanhua.sso.service;

import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.alibaba.dubbo.config.annotation.Reference;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tanhua.common.mapper.QuestionUserLockMapper;
import com.tanhua.common.mapper.UserMapper;
import com.tanhua.common.pojo.QuestionUserLock;
import com.tanhua.common.pojo.User;
import com.tanhua.dubbo.server.api.HuanXinApi;
import com.tanhua.dubbo.server.api.LogManagementApi;
import com.tanhua.sso.vo.LoginResult;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;
import org.springframework.util.DigestUtils;

import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class UserService {

    @Autowired
    private QuestionUserLockMapper questionUserLockMapper;

    @Autowired
    private UserMapper userMapper;

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    private static final String FREEZE_USER_REDIS_KEY_PREFIX = "FREEZE_USER";

    @Value("${jwt.secret}")
    private String secret;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    @Reference(version = "1.0.0")
    private HuanXinApi huanXinApi;

    @Reference(version = "1.0.0")
    private LogManagementApi logManagementApi;

    /**
     * 校验验证码的方法
     *
     * @param phone
     * @param code
     * @return
     */
    public LoginResult login(String phone, String code) {
        //在之前的方法中我们已经将用户的手机号作为key,验证码作为值存入了redis中,此时我们需要用用户手机号这个key做判断
        String redisKey = "CHECK_CODE_" + phone;
        //从redis中用key获取对应的值,其实就是用户的验证码
        String redisData = redisTemplate.opsForValue().get(redisKey);
        //判断用户输入的验证码与redis中存储的验证码是否一致
        if (!StringUtils.equals(code, redisData)) {
            //如果不一致说明验证码错误,直接返回一个null
            return null;
        }
        //验证成功之后则需要从redis中将该验证码数据删除
        redisTemplate.delete(redisKey);
        //验证码验证成功之后则需要判断用户是否已经注册
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("mobile", phone);
        //调用mapper层,使用mybatisplus提供的selectOne方法查询数据库中是否存在该用户,返回一个user对象
        User user = userMapper.selectOne(queryWrapper);
        //定义一个参数,用于对isNew属性进行赋值
        boolean flag = false;
        //如果user为null则说明是新客户
        if (null == user) {
            //创建一个user对象
            user = new User();
            user.setMobile(phone);
            //用户密码实际在系统中用不到所以设置一个初始密码使用MD5进行加密
            user.setPassword(DigestUtils.md5DigestAsHex("123456".getBytes(StandardCharsets.UTF_8)));
            //将用户存入数据库
            userMapper.insert(user);
            //给isNew属性赋值为true,代表他是新用户
            flag = true;
            //注册环信用户
            Boolean register = huanXinApi.register(user.getId());

            if (!register) {
                //注册失败记录日志
                log.error("注册环信用户失败~ userId = " + user.getId());
            }

            //给用户的灵魂表添加锁功能
            //List<QuestionUserLock> questionUserLocks = new ArrayList<>();
            QuestionUserLock questionUserLock = new QuestionUserLock();
            questionUserLock.setUserId(user.getId());
            questionUserLock.setQuestionnaireId(1);
            questionUserLock.setIsLock(0);
            questionUserLockMapper.insert(questionUserLock);
            //questionUserLocks.add(questionUserLock);

            QuestionUserLock questionUserLock1 = new QuestionUserLock();
            questionUserLock1.setUserId(user.getId());
            questionUserLock1.setQuestionnaireId(2);
            questionUserLock1.setIsLock(1);
            questionUserLockMapper.insert(questionUserLock1);
            //questionUserLocks.add(questionUserLock1)

            QuestionUserLock questionUserLock2 = new QuestionUserLock();
            questionUserLock2.setUserId(user.getId());
            questionUserLock2.setQuestionnaireId(3);
            questionUserLock2.setIsLock(1);
            questionUserLockMapper.insert(questionUserLock2);
            //questionUserLocks.add(questionUserLock2);



        }

        //判断该用户是否在冻结中
        redisKey = getFreezeUserRedisKeyPrefix(Convert.toInt(user.getId()));
        String data = redisTemplate.opsForValue().get(redisKey);
        if (ObjectUtil.isNotEmpty(data)) {
            JSONObject jsonObject = JSONUtil.parseObj(data);
            Object freezingRange = jsonObject.get("freezingRange");
            if (ObjectUtil.equal(freezingRange,1)){
                //表示用户是登录被封禁中
                return null;
            }
        }

        //LoginResult中isNew有了我们还需要一个token,无论是新老用户都需要生成一个token
        Map<String, Object> claims = new HashMap<String, Object>();
        claims.put("id", user.getId());//payload中存入用户的id
        // 生成token的方法,直接粘贴复制
        String token = Jwts.builder()
                .setClaims(claims) //payload，存放数据的位置，不能放置敏感数据，如：密码等
                .signWith(SignatureAlgorithm.HS256, secret) //设置加密方法和加密盐,盐值从配置文件中获取
                .setExpiration(new DateTime().plusHours(12).toDate()) //设置过期时间，12h后过期
                .compact();
        /*try {
            //使用rocketmq作为消息中间件,但目前暂时使用不到
            Map<String, Object> msg = new HashMap<>();
            msg.put("id", user.getId());
            msg.put("date", System.currentTimeMillis());
            rocketMQTemplate.convertAndSend("tanhua-sso-login", msg);
        } catch (MessagingException e) {
            log.error("消息发送失败!", e);
        }*/
        logManagementApi.logWrite(user.getId());

        return LoginResult.builder().token(token).isNew(flag).build();
    }

    /**
     * 校验token,根据token查询用户数据
     *
     * @param token
     * @return
     */
    public User findUserByToken(String token) {
        try {
            //解析token的方法,盐值从配置文件中获取,获取token体
            Map<String, Object> body = Jwts.parser().setSigningKey(secret)//盐值要相同
                    .parseClaimsJws(token).getBody();
            //创建一个user对象
            User user = new User();
            //之前封装的token在payload中存入了用户的id,将其赋给新的user对象,id是long类型的所以此处需要先做一下类型转换
            user.setId(Long.valueOf(body.get("id").toString()));
            //需要返回user对象中的mobile，需要查询数据库获取到mobile数据
            //如果每次都查询数据库，必然会导致性能问题，需要对用户的手机号进行缓存操作
            //数据缓存时，需要设置过期时间，过期时间要与token的时间一致
            //如果用户修改了手机号，需要同步修改redis中的数据
            //将用户的ID作为键
            String redisKey = "TANHUA_USER_MOBILE_" + user.getId();
            //如果redis中没有这个key
            if (!redisTemplate.hasKey(redisKey)) {
                //调用userMapper的mp方法在数据库中获取到这个用户
                User u = userMapper.selectById(user.getId());
                //获取该用户的手机号将其赋值给刚创建的user对象
                user.setMobile(u.getMobile());
                long timeout = Long.valueOf(body.get("exp").toString()) * 1000 - System.currentTimeMillis();
                //将用户手机号存入redis中,设置过期时间，过期时间要与token的过期时间一致
                redisTemplate.opsForValue().set(redisKey, u.getMobile(), timeout, TimeUnit.MILLISECONDS);
            } else {
                //如果redis中已经有这个key了,根据key获取redis中存储的手机号
                String mobile = redisTemplate.opsForValue().get(redisKey);
                //给新创建的user对象的手机号赋值
                user.setMobile(mobile);
            }
            //返回user
            return user;
        } catch (ExpiredJwtException e) {
            //记录日志token已过期
            log.info("token已经过期！ token = " + token);
        } catch (Exception e) {
            //记录日志token不合法
            log.error("token不合法！ token = " + token, e);
        }
        //无意义只是为了程序通过编译
        return null;
    }

    private String getFreezeUserRedisKeyPrefix(Integer uid) {
        return FREEZE_USER_REDIS_KEY_PREFIX + uid;
    }
}
